package com.wave.wavesystem.ai.test.docquery;

import org.springframework.ai.chat.prompt.PromptTemplate;
import org.springframework.ai.rag.generation.augmentation.ContextualQueryAugmenter;

/**
 * 当查询不到文档时 可以告诉ai 怎么做  上下文查询增强器
 */
public class QueryContextualAugmenterFactory {


    public static ContextualQueryAugmenter createQueryAugmenter() {
        PromptTemplate emppty = new PromptTemplate("""
                你应该输出下面的内容：
                抱歉我没有查询到内容，请联系阿威。
                """);
        return ContextualQueryAugmenter.builder()
                .promptTemplate(emppty)
                .allowEmptyContext(true)
               .build();
    }


}
